#pragma once

#include <cassert>
#include <vector>

namespace tcv {

struct ImageProcessConfig {
  bool bgr_to_rgb = false;

  /**
   * @brief 设置图像通道交换
   * @param bgr_to_rgb
   */
  void enableSwapRB() { bgr_to_rgb = true; }

  /**
   * @brief 设置归一化参数
   *
   * @param mean
   * @param std
   */
  void setNormalizationParams(const std::vector<float>& mean, const std::vector<float>& std) {
    assert(mean.size() == 3 && std.size() == 3 && "ImageProcessConfig: mean and std must be 3 elements");

  }

};

} // namespace tcv